import os
import pandas as pd

# 定义文件夹路径
data_folder = 'data'
high_temp_file = '高温.xlsx'
low_temp_file = '低温.xlsx'

# 初始化两个空的DataFrame来存储高温和低温数据
high_temp_df = pd.DataFrame(columns=['时间', '温度'])
low_temp_df = pd.DataFrame(columns=['时间', '温度'])

# 遍历data文件夹中的所有文件
for filename in os.listdir(data_folder):
    if filename.endswith('.xlsx') or filename.endswith('.xls'):
        file_path = os.path.join(data_folder, filename)
        # 读取Excel文件
        df = pd.read_excel(file_path)
        
        # 假设时间和温度列分别命名为'时间'和'温度'（根据实际情况修改）
        if '时间' in df.columns and '温度' in df.columns:
            # 筛选超过69度和低于31度的数据
            high_temp_mask = df['温度'] > 69
            low_temp_mask = df['温度'] < 31
            
            # 将筛选后的数据添加到对应的DataFrame中
            high_temp_df = pd.concat([high_temp_df, df.loc[high_temp_mask, ['时间', '温度']]])
            low_temp_df = pd.concat([low_temp_df, df.loc[low_temp_mask, ['时间', '温度']]])

# 将高温和低温数据分别保存到新的Excel文件中
high_temp_df.to_excel(high_temp_file, index=False)
low_temp_df.to_excel(low_temp_file, index=False)

print(f"高温数据已保存到 {high_temp_file}")
print(f"低温数据已保存到 {low_temp_file}")